import React from 'react';
import { Navigate } from 'react-router-dom';

// 高阶组件
const withAuthRedirect = (WrappedComponent) => {
  return (props) => {
    const token = localStorage.getItem('token'); // 检查 token

    // 如果没有 token，重定向到登录页
    if (!token) {
      return <Navigate to="/login" replace />;
    }

    // 如果有 token，渲染传入的组件
    return <WrappedComponent {...props} />;
  };
};

export default withAuthRedirect;
